package com.maomao.admin.service.system.impl;

import com.maomao.admin.common.constant.UserConstants;
import com.maomao.admin.common.exception.ServiceException;
import com.maomao.admin.common.utils.StringUtils;
import com.maomao.admin.domain.core.vo.TableDataInfo;
import com.maomao.admin.domain.system.entry.QSysPost;
import com.maomao.admin.domain.system.entry.QSysUserPost;
import com.maomao.admin.domain.system.entry.SysPost;
import com.maomao.admin.persistence.repository.SysPostRepository;
import com.maomao.admin.persistence.repository.SysUserPostRepository;
import com.maomao.admin.service.system.AbstractPageService;
import com.maomao.admin.service.system.ISysPostService;
import com.querydsl.core.types.Predicate;
import com.querydsl.jpa.impl.JPAQuery;
import com.querydsl.jpa.impl.JPAQueryFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.Arrays;
import java.util.LinkedList;
import java.util.List;
import java.util.function.Function;

/**
 * 岗位信息 服务层处理
 *
 * @author mao.kaiyin
 */
@Service
public class SysPostServiceImpl extends AbstractPageService implements ISysPostService {


    @Resource
    private JPAQueryFactory queryFactory;

    @Autowired
    private SysPostRepository postRepository;

    @Autowired
    private SysUserPostRepository userPostRepository;

    /**
     * 查询岗位信息集合
     *
     * @param post 岗位信息
     * @return 岗位信息集合
     */
    @Override
    public List<SysPost> selectPostList(SysPost post) {
        JPAQuery<SysPost> jpaQuery = this.getSysPost(post);
        return jpaQuery.fetch();
    }


    private JPAQuery<SysPost> getSysPost(SysPost post) {
        JPAQuery<SysPost> jpaQuery = queryFactory.selectFrom(QSysPost.sysPost);
        List<Predicate> predicates = new LinkedList<>();
        if (StringUtils.isNotBlank(post.getPostCode())) {
            predicates.add(QSysPost.sysPost.postCode.contains(post.getPostCode()));
        }
        if (StringUtils.isNotBlank(post.getPostName())) {
            predicates.add(QSysPost.sysPost.postName.contains(post.getPostName()));
        }
        if (StringUtils.isNotBlank(post.getStatus())) {
            predicates.add(QSysPost.sysPost.status.eq(post.getStatus()));
        }
        if (predicates.size() > 0) {
            jpaQuery.where(predicates.toArray(new Predicate[0])).fetch();
        }
        return jpaQuery;
    }

    /**
     * 查询岗位信息集合
     *
     * @param post 岗位信息
     * @return 岗位信息集合
     */
    @Override
    public TableDataInfo<SysPost> selectPostPage(SysPost post) {
        JPAQuery<SysPost> jpaQuery = this.getSysPost(post);
        return new TableDataInfo<>(jpaQuery.fetch(), jpaQuery.fetchCount());
    }

    /**
     * 查询所有岗位
     *
     * @return 岗位列表
     */
    @Override
    public List<SysPost> selectPostAll() {
        return postRepository.findAll();
    }

    /**
     * 通过岗位ID查询岗位信息
     *
     * @param postId 岗位ID
     * @return 角色对象信息
     */
    @Override
    public SysPost selectPostById(Long postId) {
        return postRepository.findById(postId).orElse(null);
    }

    /**
     * 根据用户ID获取岗位选择框列表
     *
     * @param userId 用户ID
     * @return 选中岗位ID列表
     */
    @Override
    public List<Long> selectPostListByUserId(Long userId) {
        return queryFactory.select(QSysUserPost.sysUserPost.id.postId).from(QSysUserPost.sysUserPost).where(QSysUserPost.sysUserPost.id.userId.eq(userId)).fetch();
    }

    /**
     * 校验岗位名称是否唯一
     *
     * @param post 岗位信息
     * @return 结果
     */
    @Override
    public String checkPostNameUnique(SysPost post) {
        return this.checkUnique(post, postRepository::findByPostName, post.getPostName());
    }

    private String checkUnique(SysPost post, Function<String, List<SysPost>> function, String val) {
        Long postId = StringUtils.isNull(post.getPostId()) ? -1L : post.getPostId();
        List<SysPost> list = function.apply(val);
        if (list.size() > 1) {
            return UserConstants.NOT_UNIQUE;
        }
        if (list.size() == 0) {
            return UserConstants.UNIQUE;
        }
        SysPost info = list.get(0);
        if (info.getPostId().longValue() != postId.longValue()) {
            return UserConstants.NOT_UNIQUE;
        }
        return UserConstants.UNIQUE;
    }

    /**
     * 校验岗位编码是否唯一
     *
     * @param post 岗位信息
     * @return 结果
     */
    @Override
    public String checkPostCodeUnique(SysPost post) {
        return this.checkUnique(post, postRepository::findByPostCode, post.getPostCode());
    }

    /**
     * 通过岗位ID查询岗位使用数量
     *
     * @param postId 岗位ID
     * @return 结果
     */
    @Override
    public int countUserPostById(Long postId) {
        return (int) queryFactory.selectFrom(QSysUserPost.sysUserPost).where(QSysUserPost.sysUserPost.id.postId.eq(postId)).fetchCount();
    }

    /**
     * 删除岗位信息
     *
     * @param postId 岗位ID
     * @return 结果
     */
    @Override
    public int deletePostById(Long postId) {
        postRepository.deleteById(postId);
        return 1;
    }

    /**
     * 批量删除岗位信息
     *
     * @param postIds 需要删除的岗位ID
     * @return 结果
     */
    @Override
    public int deletePostByIds(Long[] postIds) {
        for (Long postId : postIds) {
            long cnt = queryFactory.selectFrom(QSysUserPost.sysUserPost).where(QSysUserPost.sysUserPost.id.postId.eq(postId)).fetchCount();
            if (cnt > 0) {
                SysPost post = postRepository.findById(postId).get();
                throw new ServiceException(String.format("%1$s已分配,不能删除", post.getPostName()));
            }
        }
        postRepository.deleteAllById(Arrays.asList(postIds));
        return 1;
    }

    /**
     * 新增保存岗位信息
     *
     * @param post 岗位信息
     * @return 结果
     */
    @Override
    public int insertPost(SysPost post) {
        postRepository.save(post);
        return 1;
    }

    /**
     * 修改保存岗位信息
     *
     * @param post 岗位信息
     * @return 结果
     */
    @Override
    public int updatePost(SysPost post) {
        postRepository.save(post);
        return 1;
    }
}
